<script setup lang="tsx">
import {Form, FormSchema} from '@/components/Form'
import {useForm} from '@/hooks/web/useForm'
import {PropType, reactive, watch, ref, unref, nextTick} from 'vue'
import {useValidator} from '@/hooks/web/useValidator'
import {useI18n} from '@/hooks/web/useI18n'
import {ElTree, ElCheckboxGroup, ElCheckbox} from 'element-plus'
import {getMenuListApi} from '@/api/menu'
import {filter, eachTree} from '@/utils/tree'
import {findIndex} from '@/utils'
import {functionTreeApi} from "@/api/role";

const {t} = useI18n()

const {required} = useValidator()

const props = defineProps({
  currentRow: {
    type: Object as PropType<any>,
    default: () => null
  }
})

const treeRef = ref<typeof ElTree>()

const functionTreeRef = ref<typeof ElTree>()

const formSchema = ref<FormSchema[]>([
  {
    field: 'name',
    label: t('role.roleName'),
    component: 'Input'
  },
  {
    field: 'mark',
    label: t('role.mark'),
    component: 'Input'
  },
  // {
  //   field: 'status',
  //   label: t('menu.status'),
  //   component: 'Select',
  //   componentProps: {
  //     options: [
  //       {
  //         label: t('userDemo.disable'),
  //         value: 0
  //       },
  //       {
  //         label: t('userDemo.enable'),
  //         value: 1
  //       }
  //     ]
  //   }
  // },
  {
    field: 'menus',
    label: t('role.menu'),
    colProps: {
      span: 24
    },
    formItemProps: {
      slots: {
        default: () => {
          return (
              <>
                <div class="flex w-full">
                  <div class="flex-1">
                    <ElTree
                        ref={treeRef}
                        show-checkbox
                        node-key="id"
                        highlight-current
                        check-strictly
                        expand-on-click-node={false}
                        data={treeData.value}

                    >
                      {{
                        default: (data) => {
                          return <span>{data.data.meta.title}</span>
                        }
                      }}
                    </ElTree>
                  </div>
                  <div class="flex-1">
                    {unref(currentTreeData) && unref(currentTreeData)?.permissionList ? (
                        <ElCheckboxGroup v-model={unref(currentTreeData).meta.permission}>
                          {unref(currentTreeData)?.permissionList.map((v: any) => {
                            return <ElCheckbox label={v.value}>{v.label}</ElCheckbox>
                          })}
                        </ElCheckboxGroup>
                    ) : null}
                  </div>
                </div>
              </>
          )
        }
      }
    }
  },
  {
    field: 'functions',
    label: t('role.function'),
    colProps: {
      span: 24
    },
    formItemProps: {
      slots: {
        default: () => {
          return (
              <>
                <div class="flex w-full">
                  <div class="flex-1">
                    <ElTree
                        ref={functionTreeRef}
                        show-checkbox
                        node-key="id"
                        highlight-current
                        check-strictly
                        expand-on-click-node={false}
                        data={functionTreeData.value}
                    >
                      {{
                        default: (data) => {
                          return <span>{data.data.label}</span>
                        }
                      }}
                    </ElTree>
                  </div>
                </div>
              </>
          )
        }
      }
    }
  }
])

const currentTreeData = ref()
const nodeClick = (treeData: any) => {
  currentTreeData.value = treeData
}

const rules = reactive({
  name: [required()],
  role: [required()]
})

const {formRegister, formMethods} = useForm()
const {setValues, getFormData, getElFormExpose} = formMethods

const treeData = ref([])
const getMenuList = async () => {
  const res = await getMenuListApi()
  if (res) {
    treeData.value = res.data
    if (!props.currentRow) return
    await nextTick()
    const checked: any[] = []
    eachTree(props.currentRow.menus, (v) => {
      checked.push({
        id: v + '',
        permission: v.meta?.permission || []
      })
    })
    eachTree(treeData.value, (v) => {
      const index = findIndex(checked, (item) => {
        return item.id === v.id
      })
      if (index > -1) {
        const meta = {...(v.meta || {})}
        meta.permission = checked[index].permission
        v.meta = meta
      }
    })
    for (const item of checked) {
      unref(treeRef)?.setChecked(item.id, true, false)
    }
    unref(treeRef)?.setCheckedKeys(
        checked.map((v) => v.id),
        false
    )
  }
}
getMenuList()

const functionTreeData = ref([])
const getFunctionList = async () => {
  const res = await functionTreeApi()
  if (res) {
    functionTreeData.value = res.data
    if (!props.currentRow) return
    await nextTick()
    const checked: any[] = []
    eachTree(props.currentRow.functions, (v) => {
      checked.push({
        id: v + ''
      })
    })
    eachTree(functionTreeData.value, (v) => {
      const index = findIndex(checked, (item) => {
        return item.id === v.id
      })
    })
    for (const item of checked) {
      unref(functionTreeRef)?.setChecked(item.id, true, false)
    }
    unref(functionTreeRef)?.setCheckedKeys(
        checked.map((v) => v.id),
        false
    )
  }
}

getFunctionList()

const submit = async () => {
  const elForm = await getElFormExpose()
  const valid = await elForm?.validate().catch((err) => {
    console.log(err)
  })
  if (valid) {
    const formData = await getFormData()
    const checkedKeys = unref(treeRef)?.getCheckedKeys() || []
    const checkedFunctionKeys = unref(functionTreeRef)?.getCheckedKeys() || []
    let selectMenus = checkedKeys || []
    formData.menus = selectMenus
    formData.functions = checkedFunctionKeys
    console.log(formData)
    return formData
  }
}

watch(
    () => props.currentRow,
    (currentRow) => {
      if (!currentRow) return
      setValues(currentRow)
    },
    {
      deep: true,
      immediate: true
    }
)

defineExpose({
  submit
})
</script>

<template>
  <Form :rules="rules" @register="formRegister" :schema="formSchema"/>
</template>
